import java.lang.reflect.Field;
import java.sql.*;
import java.util.ArrayList;
import java.util.List;

/**
 * 泛型、反射、注解知识点的巩固
 *
 * @author zhu
 * @date 2024-04-07
 */
public class SqlRunner {
    /**
     * 连接对象
     */
    private Connection connection;

    public SqlRunner(Connection connection) {
        this.connection = connection;
    }

    /**
     * 通用的更新操作方法 - 增、删、改
     *
     * @param sql 增、删、改的SQL语句
     * @param params SQL语句需要的具体数据 - 不定长参数本质上是一个数组
     * @return 受影响行数
     */
    public int update(String sql, Object... params) {
        int row = 0;

        // 第一：定义要操作的sql语句
        // 第二：获取连接对象
        PreparedStatement ps = null;

        try {
            // 第三：预编译SQL语句
            ps = connection.prepareStatement(sql);
            // 第四：填充参数 - 循环遍历不定长参数，调用 ps.setObject(小标,值)
            setParameters(params, ps);
            // 第五：执行SQL语句
            row = ps.executeUpdate();

        } catch (SQLException e) {
            throw new RuntimeException(e);
        }finally {
            // 第六：关闭对象
            DbUtil.close(null,ps,connection);
        }
        return row;
    }

    /**
     * 填充占位符 - ctrl + alt + m
     * 1）单一职责原则
     * 2）复用性 - 一次定义，多次使用
     *
     * @param params SQL语句需要的数据
     * @param ps 连接器
     * @throws SQLException SQL异常
     */
    private static void setParameters(Object[] params, PreparedStatement ps) throws SQLException {
        for (int i = 0; i < params.length; i++) {
            ps.setObject(i + 1, params[i]);
        }
    }

    /**
     * 通过查询操作 - 查询任何数据表，返回 T 类型的 List 集合
     *
     * @param entity 实体对象
     * @param sql 要操作的SQL语句
     * @param params SQL语句需要的具体数据
     * @return T类型的List集合
     * @param <T> 实体对象的具体类型
     */
    public <T> List<T> queryList(Class<T> entity, String sql, Object... params) {
        // 第一：定义要操作的SQL语句
        // 第二：获取连接对象

        PreparedStatement ps = null;
        ResultSet rst = null;
        try {
            // 第三：预编译SQL语句
            ps = connection.prepareStatement(sql);
            // 第四：填充参数
            setParameters(params, ps);
            // 第五：执行查询对象 - ResultSet对象
            rst = ps.executeQuery();
            // 第六：把ResultSet对象中的数据封装到List<T>集合返回 - 反射、内省、注解
            return beanListHandler(rst,entity);
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }finally {
            // 第七：关闭对象
            DbUtil.close(rst,ps,connection);
        }
    }

    /**
     * 通过查询操作 - 查询任何数据表，返回 T 类型的对象
     *
     * @param entity 实体对象
     * @param sql 要操作的SQL语句
     * @param params SQL语句需要的具体数据
     * @return 返回查询的T类型的实体对象
     * @param <T> 实体对象的具体类型
     */
    public <T> T query(Class<T> entity,String sql, Object... params) {
        // 第一：定义要操作的SQL语句
        // 第二：获取连接对象

        PreparedStatement ps = null;
        ResultSet rst = null;

        try {
            // 第三：预编译SQL语句
            ps = connection.prepareStatement(sql);
            // 第四：填充参数
            setParameters(params, ps);
            // 第五：执行查询对象 - ResultSet对象
            rst = ps.executeQuery();
            // 第六：从ResultSet中读取数据返回
            return beanHandler(entity, rst);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * 循环遍历ResultSet，读取各行各列数据
     * 封装到实体对象中返回
     *
     * @param entity 实体对象的Class对象
     * @param rst 结果集对象
     * @return T类型对象
     * @param <T> 具体的实体对象类型
     */
    private <T> T beanHandler(Class<T> entity, ResultSet rst){

        // 第一：实例化实体对象
        T entityClass = null;
        try {
            entityClass = entity.getConstructor().newInstance();

            // 第二：获取结果集元数据对象
            ResultSetMetaData metaData = rst.getMetaData();
            // 第三：获取select语句后面查询的字段数量 - 循环读取各列数据
            int count = metaData.getColumnCount();
            // 第四：循环查询的各行
            if (rst.next()) {
                // 第五：循环查询的各行
                setValue(rst, entity, count, metaData, entityClass);
            }
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
        return entityClass;
    }

    /**
     * 循环遍历ResultSet，读取各行各列数据
     * 先封装到实体对象中，然后把实体对象添加到List集合中返回
     *
     * @param rst 结果集对象
     * @param entityClass 实体对象的Class对象
     * @return List<T>
     * @param <T> 具体的实体对象类型
     */
    private <T> List<T> beanListHandler(ResultSet rst,Class<T> entityClass){

        List<T> list = new ArrayList<>();

        try {
            // 第一：获取结果集元数据对象
            ResultSetMetaData metaData = rst.getMetaData();

            // 第二：获取select语句后面查询的字段数量 - 循环读取各列数据
            int columnCount = metaData.getColumnCount();
            System.out.println("查询列数：" + columnCount);

            // 第三：循环遍历结果集 - 循环各行
            while (rst.next()) {
                // 第四：实例化实体对象 - 如何赋值？
                T entity = entityClass.getConstructor().newInstance();

                // 第五：循环查询的各行
                setValue(rst, entityClass, columnCount, metaData, entity);
                // 第十二：把实体对象添加到List集合中
                list.add(entity);
            }
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
        return list;
    }

    /**
     * 循环查询的各行,把各列数据封装到实体对象中
     *
     * @param rst 结果集对象
     * @param entityClass 实体对象的Class对象
     * @param columnCount select语句后面查询的字段数量
     * @param metaData 结果集元数据对象
     * @param entity 要返回的实体对象
     * @param <T> 具体的实体对象类型
     */
    private <T> void setValue(ResultSet rst, Class<T> entityClass, int columnCount, ResultSetMetaData metaData, T entity){
        for (int i = 0; i < columnCount; i++) {
            // 第六：获取select语句后面查询的列名称
            // 可能存在列名称与实体对象属性名称不一致 - 通过Column注解进行映射 - 解析注解
            try {
                String columnName = metaData.getColumnLabel(i + 1);

                // 第七：解析注解 - 根据注解中数据表字段的名称（sta_pos），获取对应实体对象的属性的名称（staPos）
                String fieldName = getFieldName(columnName, entityClass);

                // 第八：获取各列数据
                Object columnValue = rst.getObject(i + 1);

                // 第九：获取实体对象对应的属性的Field对象
                Field field = entityClass.getDeclaredField(fieldName);

                // 第十：设置访问权限
                field.setAccessible(true);

                // 第十一：把各列数据封装到实体对象中
                field.set(entity, columnValue);

            } catch (Exception e) {
                throw new RuntimeException(e);
            }
        }
    }

    /**
     * 根据数据表字段名称，获取对应实体对象的属性名称
     * 如：sta_pos -> staPos
     *
     * @param columnLabel 数据表字段名称
     * @param entityClass 实体对象的Class对象
     * @return 实体对象的属性名称
     * @param <T> T类型的实体对象
     */
    private <T> String getFieldName(String columnLabel, Class<T> entityClass) {
        // 第一：获取实体对象的所有字段列表
        Field[] fields = entityClass.getDeclaredFields();

        // 第二：循环遍历字段列表
        for (Field field : fields) {
            // 第三：判断字段上是否存在Column注解
            if (field.isAnnotationPresent(Column.class)) {
                // 第四：获取注解的值
                String value = field.getAnnotation(Column.class).value();

                // 第五：匹配Column注解的值是否与数据表字段名称一样
                if(value != null && value.equals(columnLabel)){
                    // 第六：如果一样：则返回对应字段对象的名称（实体对象的属性名称）
                    return field.getName();
                }
            }
        }
        // 第七：如果不一样，则返回数据表字段名称
        return columnLabel;
    }

    /**
     * 查询单行单列数据
     *
     * @param sql 要操作的SQL语句
     * @param params SQL语句需要的具体数据
     * @return 查询结果
     */
    public Object querySingleValue(String sql,Object... params) {
        // 第一：定义要操作的SQL语句
        // 第二：获取连接对象

        PreparedStatement ps = null;
        ResultSet rst = null;
        Object result = null;
        try {
            // 第三：预编译SQL语句
            ps = connection.prepareStatement(sql);
            // 第四：填充参数
            setParameters(params, ps);
            // 第五：执行查询对象 - ResultSet对象
            rst = ps.executeQuery();
            // 第六：获取数据并封装
            if (rst.next()) {
                // 获取第一列数据
                result = rst.getObject(1);
            }
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            // 关闭连接
            DbUtil.close(rst, ps, connection);
        }
        return result;
    }
}
